from .ACDC import get_acdc_loader
from .LIDC import get_lidc_loader
from .ISIC import get_isic_loader

def get_loader(args):
    if args.datasets == "acdc":
        train_loader,test_loader = get_acdc_loader(root=args.data_path, batch_size=args.batch_size,train_crop_size=args.train_crop_size)
    elif args.datasets == "lidc":
        train_loader,test_loader = get_lidc_loader(root=args.data_path, batch_size=args.batch_size,train_crop_size=args.train_crop_size)
    elif args.datasets == "isic":
        train_loader,test_loader = get_isic_loader(root=args.data_path, batch_size=args.batch_size,train_crop_size=args.train_crop_size)
    else:
        raise NotImplementedError

    return train_loader,  test_loader
